from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension


setup(
    name="lzs_cuda_test",
    version="2.3.3", 
    description="CUDA kernel fusion exercise", 
    long_description="CUDA kernel fusion exercise of Spark project", 
    url="http://zgsb.com", 
    author="Xiaobao Li", 
    author_email="caibozongzi@163.com", 
    ext_modules=[
        CUDAExtension("lzs_cuda_test", ["my_test.cpp", 
                                        "my_test_2_kernel.cu"])
    ], 
    cmdclass={
        "build_ext": BuildExtension
    }
)

